package _binary_tree

import common.TreeNode
import org.junit.Assert
import org.junit.Test
import common.printLevelOrder3

/**
 * https://leetcode.cn/problems/insufficient-nodes-in-root-to-leaf-paths/description
 *
 * ```
 * 1080. 根到叶路径上的不足节点
 * 给你二叉树的根节点 root 和一个整数 limit ，请你同时删除树中所有 不足节点 ，并返回最终二叉树的根节点。
 *
 * 假如通过节点 node 的每种可能的 “根-叶” 路径上值的总和全都小于给定的 limit，则该节点被称之为 不足节点 ，需要被删除。
 *
 * 叶子节点，就是没有子节点的节点。
 *
 * 示例 1：
 * 输入：root = [1,2,3,4,-99,-99,7,8,9,-99,-99,12,13,-99,14], limit = 1
 * 输出：[1,2,3,4,null,null,7,8,9,null,14]
 *
 * 示例 2：
 * 输入：root = [5,4,8,11,null,17,4,7,1,null,null,5,3], limit = 22
 * 输出：[5,4,8,11,null,17,4,7,null,null,null,5]
 *
 * 示例 3：
 * 输入：root = [1,2,-3,-5,null,4,null], limit = -1
 * 输出：[1,null,-3,4]
 *
 * 提示：
 * 树中节点数目在范围 [1, 5000] 内
 * -105 <= Node.val <= 105
 * -109 <= limit <= 109
 * ```
 */
class leetcode_1080 {
    @Test
    fun test_3() {
        val n1 = TreeNode(1)
        val n2 = TreeNode(2)
        val n3 = TreeNode(-3)
        val n4 = TreeNode(4)
        val n5 = TreeNode(-5)
        val root = n1

        n1.left = n2
        n1.right = n3

        n2.left = n5

        n3.left = n4
        val actual = sufficientSubset(root, -1)
        val expected = arrayListOf(1, null, -3, 4)
        Assert.assertEquals(expected.toString(), printLevelOrder3(actual).toString())
    }

    private fun sufficientSubset(root: TreeNode?, limit: Int): TreeNode? {
        /**
        题型：二叉树的后序遍历 + 分治法

        思路：
        当为叶子节点时，若累加sum > limit, 保留该叶子节点
        遍历左右子树后，若左右子树都被删除，那么当前节点也要删除。

        https://leetcode.cn/problems/insufficient-nodes-in-root-to-leaf-paths/solutions/8726/hou-xu-bian-li-python-dai-ma-java-dai-ma-by-liweiw
        https://www.cnblogs.com/tianyiya/p/17426000.html
         */
        return delete(root, limit, 0)
    }

    private fun delete(root: TreeNode?, limit: Int, sum: Int): TreeNode? {
        // 1 确定函数参数以及返回值
        // 2 确定终止条件
        if (null == root) {
            return null
        }
        // 收集结果 ： 当叶子节点时，若sum > limit,则保留叶子节点
        if (root.left == null && root.right == null) {
            if (root.`val` + sum < limit) {
                return null
            } else {
                return root
            }
        }

        // 3 确定单层递归的处理逻辑:后序遍历
        val newSum = sum + root.`val`
        // 左
        root.left = delete(root.left, limit, newSum)

        // 右
        root.right = delete(root.right, limit, newSum)

        // 中
        // 因为左右子树都被删除了，那么这个节点也没有保留的必要了。
        if (root.left == null && root.right == null) {
            return null
        }

        return root
    }
}